from mindmaterial.e3 import TensorProduct, Linear, FullyConnectedTensorProduct

from mindmaterial.e3 import Irreps

input1 = Irreps('32x0e')
input2 = Irreps('2x0e')
out = Irreps('64x0e+32x1o')

print(input1, input2, out)

tp = FullyConnectedTensorProduct(input1, input2, out)

print(tp)

from e3nn.o3 import FullyConnectedTensorProduct as FTP2
from e3nn.o3 import Irreps

input1 = Irreps('32x0e')
input2 = Irreps('2x0e')
out = Irreps('64x0e+32x1o')


tp2 = FTP2(input1, input2, out)

print(tp2)
